package com.qf.utils;

import java.sql.Connection;

/**
 * 事务工具类
 * @author 千锋教育
 * @Company http://www.mobiletrain.org/
 * @Version 1.0
 */
public class TxUtils {

    private static ThreadLocal<Connection> tl = new ThreadLocal<>();

    /**
     * 获取连接对象
     *
     * @return
     */
    public static Connection getConn(){
        //获取连接对象
        Connection conn = tl.get();
        //如果对象为null
        if(conn == null) {
            //创建连接对象
            conn = DataSourceUtils.getConnection();
            tl.set(conn);
        }

        return conn;
    }

    /**
     * 添加连接对象
     */
    public static void set(Connection conn){
        tl.set(conn);
    }

    /**
     * 移除线程中的对象
     */
    public static void remove(){
        tl.remove();
    }

}
